import torch
from torch import nn

# ten words  2D vector
embedding = torch.nn.Embedding(10,2)
# three  sentences   two words
input = torch.arange(0,6).view(3,2).long()
input = torch.autograd.Variable(input)
output = embedding(input)
print(output.size())
print(embedding.weight.size())